import util.tool as tool
from lib.llm import invoke

# summary = tool.read_file("data/摘要分隔.json")
summary = tool.read_file("data/摘要.md")


# 示例
example = """
"""

# 你是一名资深前端开发工程师，负责将文案描述转换为结构化JSON数据，用于后续的自动渲染。请根据我提供的内容描述，转换成JSON

system_template = """
请将提供的文档内容转换为结构化的JSON格式，要求保留关键信息并建立合理的层级关系，以便后续可视化使用。

## 要求：
JSON字段使用纯中文
"""


# 对话模型调用
result = invoke(
    system_template=system_template,
    user_template="{summary}",
    input_data={"summary": summary, "example": example},
)

print("result", result)
# print("json", tool.extract_json_from_str2(result))
json = tool.extract_json_from_str2(result)

# 保存结果json
tool.save_json("data/2-网页转JSON.json", json)
print("保存结果到文件 2-网页转JSON.json")
